package main

import (
	"GO/common"
	"GO/tools"
	"os"

	"github.com/gin-gonic/gin"
	"github.com/spf13/viper"
)

func main() {
	InitConfig()
	db := common.InitDB()
	defer db.Close()
	// 创建一个默认的路由引擎
	r := gin.Default()
	r.Use(tools.LoggerToFile())
	// 获取一个路由树
	r = CollectRouter(r)
	//获取配置文件里的端口号
	port := viper.GetString("server.port")
	if port != "" {
		r.Run(":" + port)
	}
}

func InitConfig() {
	//获取当前的工作目录
	workDir, _ := os.Getwd()
	//设置读取的文件名
	viper.SetConfigName("config")
	//设置读取的文件类型
	viper.SetConfigType("yml")
	//设置文件的路径
	viper.AddConfigPath(workDir + "/config")
	err := viper.ReadInConfig()
	if err != nil {
		panic("Failed to read the configuration file,err" + err.Error())
	}
}
